package com.algorithm;

/**
 * @author: sun.hongliang
 * @create: 2019/07/04 10:29
 */
public class MergeTwoLinkList {
    public static void main(String[] args) {
        final ListNode listNode1 = initListNode();
        final ListNode listNode2 = initListNode();
        final ListNode listNode = mergeTwoLists(listNode1, listNode2);
        LinkListPrint.printListNode(listNode);
    }

    public static ListNode mergeTwoLists(ListNode l1, ListNode l2) {
        ListNode head = new ListNode(0);
        ListNode h = head;
        ListNode p = l1;
        ListNode q = l2;
        while (q != null || p != null) {
            if (q == null && p != null) {
                h.next = new ListNode(p.val);
                p = p.next;
            } else if (p == null && q != null) {
                h.next = new ListNode(q.val);
                q = q.next;
            } else if (p != null && q != null) {
                if (p.val < q.val) {
                    h.next = new ListNode(p.val);
                    p = p.next;
                } else {
                    h.next = new ListNode(q.val);
                    q = q.next;
                }
            }
            h = h.next;
        }
        return head.next;
    }

    public static ListNode mergeTwoLists2(ListNode l1, ListNode l2) {
        ListNode head = new ListNode(-1);
        ListNode h = head;
        ListNode p = l1;
        ListNode q = l2;
        while (p != null && q != null) {
            if (p.val <= q.val) {
                h.next = p;
                p = p.next;
            } else {
                h.next = q;
                q = q.next;
            }
            h = h.next;
        }
        h.next = p == null ? q : p;
        return head.next;
    }

    public static ListNode initListNode() {
        ListNode head = new ListNode(0);
        ListNode one = new ListNode(1);
        ListNode two = new ListNode(2);
        head.next = one;
        one.next = two;
        return head;
    }
}
